import pickle
import os
from stock.utils.get_stock_price import get_stock_price

def get_stock_data(key):
    # 本地文件名
    filename = f"stock_data_{key}.pkl"

    # 尝试从本地加载
    if os.path.exists(filename):
        with open(filename, 'rb') as f:
            return pickle.load(f)

    # 如果本地没有，则从远程获取（这里需要替换为实际的远程获取逻辑）
    stock_list = fetch_remote_stock_data(key)  # 假设这是你的远程获取函数

    # 保存到本地文件
    with open(filename, 'wb') as f:
        pickle.dump(stock_list, f)

    return stock_list

# 示例远程获取函数（需要根据实际情况实现）
def fetch_remote_stock_data(code):
    # 假设我们有一些示例数据
    # code = '101.HG00Y'
    klt = '101'
    count = '50'
    end = '20501010'
    list1 = get_stock_price(code, klt, count, end)
    return list1
